﻿#include "page3.h"
#include "SearchDialog.h"
#include "AddInventoryDialog.h"
#include "databaseManager.h"
#include <QVBoxLayout>
#include <QPixmap>
#include <QPalette>
#include <QMessageBox>
#include <QWidget>
#include <QTableWidget>
#include <QSqlTableModel>
#include <QSqlRecord>
#include <QSqlField>
#include <QtCharts/QChartView>
#include <QtCharts/QPieSeries>
#include <QtCharts/QPieSlice>
#include <QMouseEvent>
#include <QtCharts/QPieLegendMarker>
#include <QLabel>

QT_CHARTS_USE_NAMESPACE

Page3::Page3(QWidget *parent) :
    QWidget(parent),
    mainLayout(new QVBoxLayout(this)),
    chartView(nullptr)
{
    // 设置窗口大小
    setFixedSize(1280, 720);
    //设置背景
    QPixmap background(":new/prefix1/bg.jpg");
    QLabel* backgroundLabel = new QLabel(this);
    backgroundLabel->setPixmap(background.scaled(this->size())); // 缩放背景图片以适应窗口大小

    backgroundLabel->setGeometry(0, 0, this->width(), this->height());
    backgroundLabel->lower();

    // 设置按钮的初始Y轴位置
    int yPos = 222;

    // Assuming you have a QIcon object created from your image file
    QIcon icon1(":new/prefix1/order.jpg");
    QIcon icon11(":new/prefix1/order2.jpg");
    // Create the QPushButton and set its properties
    orderButton = new QPushButton(this);
    orderButton->setIcon(icon1);  // Set the icon
    orderButton->setIconSize(QSize(260, 71));  // Set the icon size
    orderButton->setFlat(true);  // Make the button flat (no background)
    orderButton->setFixedSize(260, 71);  // Set the fixed size of the button
    orderButton->setGeometry(1, yPos, 260, 71);
    connect(orderButton, &QPushButton::clicked, this, [this](){
        emit orderRequested();
    });
    connect(orderButton, &QPushButton::pressed, this, [this, icon11]() {
        orderButton->setIcon(icon11);  // Change to icon10 when pressed
    });

    connect(orderButton, &QPushButton::released, this, [this, icon1]() {
        orderButton->setIcon(icon1);  // Revert to icon1 when released
        emit orderRequested();  // Emit the signal after the button is released
    });

    // 调整下一个按钮的Y轴位置
    yPos += 68;

    // 创建并设置生产按钮
    QIcon icon2(":new/prefix1/production.jpg");
    QIcon icon12(":new/prefix1/production2.jpg");
    productionButton = new QPushButton("", this);
    productionButton->setIcon(icon2);  // Set the icon
    productionButton->setIconSize(QSize(260, 71));  // Set the icon size
    productionButton->setFlat(true);  // Make the button flat (no background)
    productionButton->setFixedSize(260, 71);  // Set the fixed size of the button
    productionButton->setGeometry(1, yPos, 260, 71);
    connect(productionButton, &QPushButton::clicked, this, [this]() {
        emit productionRequested();
    });
    connect(productionButton, &QPushButton::pressed, this, [this, icon12]() {
        productionButton->setIcon(icon12);
    });
    connect(productionButton, &QPushButton::released, this, [this, icon2]() {
        productionButton->setIcon(icon2);
        emit productionRequested();  // Emit the signal after the button is released
    });

    // 调整下一个按钮的Y轴位置
    yPos += 68;

    // 创建并设置财务按钮
    QIcon icon3(":new/prefix1/finance.jpg");
    QIcon icon13(":new/prefix1/finance2.jpg");
    financeButton = new QPushButton("", this);
    financeButton->setIcon(icon3);  // Set the icon
    financeButton->setIconSize(QSize(260, 71));  // Set the icon size
    financeButton->setFlat(true);  // Make the button flat (no background)
    financeButton->setFixedSize(260, 71);  // Set the fixed size of the button
    financeButton->setGeometry(1, yPos, 260, 71);
    connect(financeButton, &QPushButton::clicked, this, [this]() {
        emit financeRequested();
    });
    connect(financeButton, &QPushButton::pressed, this, [this, icon13]() {
        financeButton->setIcon(icon13);
    });
    connect(financeButton, &QPushButton::released, this, [this, icon3]() {
        financeButton->setIcon(icon3);
        emit financeRequested();  // Emit the signal after the button is released
    });

    // 调整下一个按钮的Y轴位置
    yPos += 68;

    // 创建并设置库存按钮
    QIcon icon4(":new/prefix1/warehouse.jpg");
    QIcon icon14(":new/prefix1/warehouse2.jpg");
    warehouseButton = new QPushButton("", this);
    warehouseButton->setIcon(icon14);  // Set the icon
    warehouseButton->setIconSize(QSize(260, 71));  // Set the icon size
    warehouseButton->setFlat(true);  // Make the button flat (no background)
    warehouseButton->setFixedSize(260, 71);  // Set the fixed size of the button
    warehouseButton->setGeometry(1, yPos, 260, 71);
    connect(warehouseButton, &QPushButton::clicked, this, [this]() {
        emit storeRequested();
    });

    connect(warehouseButton, &QPushButton::pressed, this, [this, icon4]() {
        warehouseButton->setIcon(icon4);
    });

    connect(warehouseButton, &QPushButton::released, this, [this, icon14]() {
        warehouseButton->setIcon(icon14);
        emit storeRequested();  // Emit the signal after the button is released
    });


    inventoryButton = new QPushButton(u8"", this);
    inventoryButton->setGeometry(265, 60, 264, 75); // 调整此位置以适应屏幕
    inventoryButton->setStyleSheet("QPushButton {"
                             " border: none; "
                             "background-image: url(:new/prefix1/kucun.jpg); "
                             "background-repeat: no-repeat; "
                             "background-position: center; }");

    // Connect pressed signal to change the background image to shouzhibiao2.jpg
    connect(inventoryButton, &QPushButton::pressed, this, [this]() {
        inventoryButton->setStyleSheet("QPushButton {"
                                       " border: none; "
                                       "background-image: url(:/new/prefix1/kucun2.jpg); "
                                       "background-repeat: no-repeat; "
                                       "background-position: center; }");
    });

    // Connect released signal to change the background image back to shouzhibiao.jpg
    connect(inventoryButton, &QPushButton::released, this, [this]() {
        inventoryButton->setStyleSheet("QPushButton {"
                                       " border: none; "
                                       "background-image: url(:/new/prefix1/kucun.jpg); "
                                       "background-repeat: no-repeat; "
                                       "background-position: center; }");
    });

    purchaseButton = new QPushButton(u8"", this);
    purchaseButton->setGeometry(530, 60, 264, 75); // 调整此位置以适应屏幕
    purchaseButton->setStyleSheet("QPushButton {"
                             " border: none; "
                             "background-image: url(:new/prefix1/ruku.jpg); "
                             "background-repeat: no-repeat; "
                             "background-position: center; }");    // Connect pressed signal to change the background image to shouzhibiao2.jpg
    connect(purchaseButton, &QPushButton::pressed, this, [this]() {
        purchaseButton->setStyleSheet("QPushButton {"
                                       " border: none; "
                                       "background-image: url(:/new/prefix1/ruku2.jpg); "
                                       "background-repeat: no-repeat; "
                                       "background-position: center; }");
    });

    // Connect released signal to change the background image back to shouzhibiao.jpg
    connect(purchaseButton, &QPushButton::released, this, [this]() {
        purchaseButton->setStyleSheet("QPushButton {"
                                       " border: none; "
                                       "background-image: url(:/new/prefix1/ruku.jpg); "
                                       "background-repeat: no-repeat; "
                                       "background-position: center; }");
    });

    queryButton = new QPushButton(u8"", this);
    queryButton->setGeometry(795, 60, 264, 75); // 调整此位置以适应屏幕
    queryButton->setStyleSheet("QPushButton {"
                             " border: none; "
                             "background-image: url(:new/prefix1/chaxun.jpg); "
                             "background-repeat: no-repeat; "
                             "background-position: center; }");
    // Connect pressed signal to change the background image to shouzhibiao2.jpg
    connect(queryButton, &QPushButton::pressed, this, [this]() {
        queryButton->setStyleSheet("QPushButton {"
                                       " border: none; "
                                       "background-image: url(:/new/prefix1/chaxun2.jpg); "
                                       "background-repeat: no-repeat; "
                                       "background-position: center; }");
    });

    // Connect released signal to change the background image back to shouzhibiao.jpg
    connect(queryButton, &QPushButton::released, this, [this]() {
        queryButton->setStyleSheet("QPushButton {"
                                       " border: none; "
                                       "background-image: url(:/new/prefix1/chaxun.jpg); "
                                       "background-repeat: no-repeat; "
                                       "background-position: center; }");
    });


    // 创建返回按钮
        frontButton = new QPushButton(u8"", this);
        frontButton->setGeometry(1050,0,170,55); // 调整此位置以适应屏幕
        frontButton->setStyleSheet("QPushButton {"
                                 " border: none; "
                                 "background-image: url(:new/prefix1/tuichudenglu.jpg); "
                                 "background-repeat: no-repeat; "
                                 "background-position: center; }");


        connect(frontButton, &QPushButton::pressed, this, [this]() {
            frontButton->setStyleSheet("QPushButton {"
                                     " border: none; "
                                     "background-image: url(:/new/prefix1/tuichudenglu2.jpg); "
                                     "background-repeat: no-repeat; "
                                     "background-position: center; }");
        });
        connect(frontButton, &QPushButton::released, this, [this]() {
            frontButton->setStyleSheet("QPushButton {"
                                     " border: none; "
                                     "background-image: url(:/new/prefix1/tuichudenglu.jpg); "
                                     "background-repeat: no-repeat; "
                                     "background-position: center; }");
        });

   connect(frontButton, &QPushButton::clicked, this, &Page3::onFrontButtonClicked);
   QVBoxLayout* backLayout = new QVBoxLayout(this);
   setLayout(backLayout);

    yPos += 68;
    QIcon icon6(":new/prefix1/operationGuide.jpg");
    QIcon icon16(":new/prefix1/operationGuide2.jpg");
    helpButton = new QPushButton("", this);
    helpButton->setIcon(icon6);  // Set the icon
    helpButton->setIconSize(QSize(260, 71));  // Set the icon size
    helpButton->setFlat(true);  // Make the button flat (no background)
    helpButton->setFixedSize(260, 71);  // Set the fixed size of the button
    helpButton->setGeometry(1, yPos, 260, 71);

    //连接按钮
         yPos += 68;
        QIcon icon7(":new/prefix1/connect.jpg");
        QIcon icon17(":new/prefix1/connect2.jpg");
        connectButton = new QPushButton("", this);
        connectButton->setIcon(icon7);  // Set the icon
        connectButton->setIconSize(QSize(260, 71));  // Set the icon size
        connectButton->setFlat(true);  // Make the button flat (no background)
        connectButton->setFixedSize(260, 71);  // Set the fixed size of the button
        connectButton->setGeometry(1, yPos, 260, 71);

        connect(connectButton, &QPushButton::clicked, this, [this]() {
            emit pageRequested13();
        });

        yPos += 68;

        // 创建并设置添加员工按钮
        QIcon icon5(":new/prefix1/tianjiayuangong01.jpg");
        QIcon icon15(":new/prefix1/tianjiayuangong02.jpg");
        addEmployeeButton = new QPushButton("", this);
        addEmployeeButton->setIcon(icon5);  // Set the icon
        addEmployeeButton->setIconSize(QSize(260, 71));  // Set the icon size
        addEmployeeButton->setFlat(true);  // Make the button flat (no background)
        addEmployeeButton->setFixedSize(260, 71);  // Set the fixed size of the button
        addEmployeeButton->setGeometry(1, yPos, 260, 71);

        connect(addEmployeeButton, &QPushButton::clicked, this, [this]() {
            emit addEmployeeRequested();
        });





        connect(connectButton, &QPushButton::pressed, this, [this, icon17]() {
             connectButton->setIcon(icon17);
        });
        connect(connectButton, &QPushButton::released, this, [this, icon7]() {
            connectButton->setIcon(icon7);
            emit pageRequested13();  // Emit the signal after the button is released
        });


    connect(helpButton, &QPushButton::clicked, this, [this]() {
        emit pageRequested10();
    });

    connect(helpButton, &QPushButton::pressed, this, [this, icon16]() {
         helpButton->setIcon(icon16);
    });
    connect(helpButton, &QPushButton::released, this, [this, icon6]() {
        helpButton->setIcon(icon6);
        emit pageRequested10();  // Emit the signal after the button is released
    });

    // 连接按钮的信号和槽
    connect(inventoryButton, &QPushButton::clicked, this, [this]() {
        emit pageRequested7();
    });

    connect(purchaseButton, &QPushButton::clicked, this, [this]() {
        emit openAddInventoryDialog();
    });

    connect(queryButton, &QPushButton::clicked, this, [this]() {
        emit pageRequested9();
    });

    connect(productionButton, &QPushButton::clicked, this, [this]() {
        emit productionRequested();
    });

    connect(orderButton, &QPushButton::clicked, this, [this]() {
        emit orderRequested();
    });

    connect(financeButton, &QPushButton::clicked, this, [this]() {
        emit financeRequested();
    });

    connect(warehouseButton, &QPushButton::clicked, this, [this]() {
        emit storeRequested();
    });

    setLayout(mainLayout);
}

Page3::~Page3()
{
    delete chartView;
    delete mainLayout;
}

void Page3::initializeInventoryChart()
{
    // 清除旧的图表视图
    if (chartView) {
        mainLayout->removeWidget(chartView);
        delete chartView;
    }

    // 获取库存数据列表
    QList<QPair<QString, int>> inventoryList = DatabaseManager::instance().getInventoryForPiePlot();

    // 创建饼状图系列
    QPieSeries* series = new QPieSeries();

    // 计算总量
    int totalAmount = 0;
    for (const auto& item : inventoryList) {
        totalAmount += item.second;
    }

    if (totalAmount == 0) {
        // 当库存数量为零时，创建一个空心的圆盘
        series->append("无库存，请及时添加", 1);
        QPieSlice* slice = series->slices().at(0);
        slice->setBrush(Qt::NoBrush);
        slice->setLabelVisible(true);
        slice->setLabel("所有种类库存为均0，请及时添加");
    } else {
        // 添加数据到饼状图系列，并设置标签
        for (const auto& item : inventoryList) {
            QString materialType = item.first;
            int amount = item.second;

            // 使用正确的物品名称添加切片到系列
            QPieSlice* slice = series->append(materialType, amount);

            // 计算百分比并设置标签
            slice->setLabel(QString("%1: %2%").arg(materialType).arg(static_cast<double>(amount) / totalAmount * 100, 0, 'f', 1));

            // 设置切片标签可见
            slice->setLabelVisible(true);
        }
    }

    // 创建饼状图
    QChart* chart = new QChart();
    chart->addSeries(series);
    chart->setTitle("库存物品种类及数量分布");

    // 设置图例显示在右侧
    chart->legend()->setVisible(true);
    chart->legend()->setAlignment(Qt::AlignRight);

    // 设置饼状图的动画效果
    chart->setAnimationOptions(QChart::AllAnimations);

    // 创建图表视图
    chartView = new QChartView(chart);
    chartView->setRenderHint(QPainter::Antialiasing);

    // 设置布局
    mainLayout->setContentsMargins(265, 135, 0, 0); // 设置外边距以控制图表位置

    // 添加图表视图到布局
    mainLayout->addWidget(chartView);
}


void Page3::financeHide()
{
    financeButton->hide();
}

void Page3::openAddInventoryDialog() {
    AddInventoryDialog dialog(this);
    if (dialog.exec() == QDialog::Accepted) {
        // Use the getter functions to obtain the values from the dialog
        QString rawMaterialType = dialog.getKindComboBox()->currentText();
        int amount = dialog.getQuantityInput()->text().toInt();
        int price = dialog.getPriceInput()->text().toInt();
        QString date = dialog.getDateEdit()->date().toString("yyyy-MM-dd");
        int totalprice = price * amount;
        QList<int> financeTotals = DatabaseManager::instance().getFinanceLastTotal();
        int totalIncome = financeTotals[0];
        int totalOutcome = financeTotals[1] + totalprice;
        int totalInterest = totalIncome - totalOutcome;
        int id = DatabaseManager::instance().getLastEnterRecordId();
        QString type = "支出";
        // 插入库存到数据库
        if (DatabaseManager::instance().changeInventory(rawMaterialType, amount) &&
            DatabaseManager::instance().insertEnterRecord(date, rawMaterialType, amount, price, totalprice) &&
            DatabaseManager::instance().insertFinanceRecord(id, type, date, totalprice, totalIncome, totalOutcome, totalInterest)) {
            QMessageBox::information(this, u8"成功", u8"库存已成功添加到数据库");
        } else {
            QMessageBox::warning(this, u8"失败", u8"添加库存到数据库失败");
        }
    }
}

void Page3::signuphide() {
    addEmployeeButton->hide();
}
void Page3::onFrontButtonClicked()
{
    QMessageBox msgBox;
    msgBox.setText("确定退出登录吗?");
    msgBox. setWindowTitle("退出登录");
    msgBox.setStandardButtons(QMessageBox::Ok | QMessageBox::Cancel);
    msgBox.button(QMessageBox::Ok)->setText("确认");
    msgBox.button(QMessageBox::Cancel)->setText("取消");
    msgBox.setStyleSheet(
        "QMessageBox {"
        "  background-color: #FFFFFF;"
        "}"
        "QMessageBox QLabel {"
        "  font-size: 18px;"
        "  font-weight: bold;"
        "  color: black;"
        "}"
        "QMessageBox QPushButton {"
        "  font-size: 18px;"
        "  font-weight: bold;"
        "  border: 2px solid #E0E0E0;"
        "  border-radius: 20px;"
        "  padding: 10px;"
        "  background-color: #0093D1;"
        "  color: white;"
        "}"
    );

    if (msgBox.exec() == QMessageBox::Ok) {
        emit pageRequested1();
    }
}
